import logging

from django import http
from django.core.cache import cache
# from django.shortcuts import render
from django.views import View

from meiduo_mall.utils.response_code import RETCODE
from .models import Area

# Create your views here.

logger = logging.getLogger('django')


class AreasView(View):
    """省市区三级联动"""

    def get(self, request):
        """查询省市区数据"""
        area_id = request.GET.get("area_id")

        if not area_id:
            # 查询省级数据

            # 读取省级数据缓存
            province_list = cache.get('province_list')

            if not province_list:
                try:
                    # 查询省级单位
                    parent_model_list = Area.objects.filter(parent__isnull=True)
                    # 序列化省级数据信息
                    province_list = []
                    # 遍历模型类列表，拿到其中的每一条数据
                    for province_model in parent_model_list:
                        parent_dict = {
                            'id': province_model.id,
                            'name': province_model.name,
                        }
                        # 将数据添加到列表中
                        province_list.append(parent_dict)

                    # 存储省级数据缓存
                    cache.set('province_list', province_list, 3600)

                except Exception as e:
                    logger.error(e)
                    return http.JsonResponse({'code': RETCODE.DBERR, 'errmsg': "省级数据错误"})

            # 返回响应
            return http.JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'province_list': province_list})

        else:  # 市区级数据信息
            # 读取市区级数据缓存信息
            sub_data = cache.get('sub_area_' + area_id)

            if not sub_data:
                try:
                    # 查询城市或者市县级数据
                    parent_model = Area.objects.get(id=area_id)

                    # 将模型列表转化为字典列表
                    sub_model_list = parent_model.subs.all()

                except Exception as e:
                    logger.error(e)
                    return http.JsonResponse({'code': RETCODE.DBERR, 'errmsg': '市区级数据错误'})
                else:
                    # 将子级模型列表转化为字典列表
                    sub_list = []
                    for sub_model in sub_model_list:
                        sub_dict = {
                            'id': sub_model.id,
                            'name': sub_model.name
                        }
                        sub_list.append(sub_dict)

                    # 构造子级的JSON数据
                    sub_data = {
                        'id': parent_model.id,
                        'name': parent_model.name,
                        'subs': sub_list,

                    }
                    cache.set('sub_area_' + area_id, sub_data, 3600)
            # 响应城市或市的json数据
            return http.JsonResponse({'code': RETCODE.OK, 'errmsg': 'OK', 'sub_data': sub_data})
